package com.xzz.stack;

import java.util.Arrays;
import java.util.Stack;

/**
 * @author: hhz
 * @create: 2021-12-29 16:11
 * 503. 下一个更大元素 II
 * 给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。
 * 输入: [1,2,1]
 * 输出: [2,-1,2]
 **/
public class NextGreaterElements {
    public static void main(String[] args) {
        System.out.println(Arrays.toString(nextGreaterElements(new int[]{1,2,1})));
        System.out.println(Arrays.toString(nextGreaterElements2(new int[]{5,4,3,2,1})));
    }
    public static int[] nextGreaterElements(int[] nums) {
        int[] tem = new int[2*nums.length];
        for (int i = 0; i < tem.length; i++) {
            if(i<nums.length){
                tem[i] = nums[i];
            }else {
                tem[i] = nums[i-nums.length];
            }

        }
        //
        int[] res = new int[tem.length];
        Stack<Integer> stack = new Stack<>();
        for (int i = tem.length-1; i >=0; i--) {
            while (!stack.empty()&&stack.peek()<=tem[i]){
                stack.pop();
            }
            res[i]=stack.empty()?-1:(stack.peek());
            stack.push(tem[i]);
        }
        return Arrays.copyOf(res,nums.length);
    }
    public static int[] nextGreaterElements2(int[] nums) {

        int[] res = new int[nums.length];
        for (int i = 0; i < res.length; i++) {
            res[i]=-1;
        }
        Stack<Integer> stack = new Stack<>();
        //正着入栈
        for (int i = 0; i < 2*nums.length-1; i++) {
            while (!stack.empty() && nums[i%nums.length]>nums[stack.peek()]){
                res[stack.pop()] = nums[i%nums.length];
            }
            if(i<nums.length){
                stack.push(i);
            }

        }
        return res;
    }
}
